/*
 * contrib/xml2/xslt_proc.c
 *
 * XSLT 处理函数（需要 libxslt）
 *
 * John Gray，Torchbox 2003-04-01
 */
#include "postgres.h"

#include "executor/spi.h"
#include "fmgr.h"
#include "funcapi.h"
#include "miscadmin.h"
#include "utils/builtins.h"
#include "utils/xml.h"

#ifdef USE_LIBXSLT

/* libxml包含 */

#include <libxml/xpath.h>
#include <libxml/tree.h>
#include <libxml/xmlmemory.h>

/* libxslt 包含 */

#include <libxslt/xslt.h>
#include <libxslt/xsltInternals.h>
#include <libxslt/security.h>
#include <libxslt/transform.h>
#include <libxslt/xsltutils.h>
#endif							/* USE_LIBXSLT */


#ifdef USE_LIBXSLT

/* 从 xpath.c 引入的声明 */
extern PgXmlErrorContext *pgxml_parser_init(PgXmlStrictness fc_strictness);

/* 本地定义 */
static const char **fc_parse_params(text *fc_paramstr);
#endif							/* USE_LIBXSLT */


PG_FUNCTION_INFO_V1(xslt_process);

Datum xslt_process(PG_FUNCTION_ARGS)
{
#ifdef USE_LIBXSLT

	text	   *fc_doct = PG_GETARG_TEXT_PP(0);
	text	   *fc_ssheet = PG_GETARG_TEXT_PP(1);
	text	   *fc_result;
	text	   *fc_paramstr;
	const char **fc_params;
	PgXmlErrorContext *fc_xmlerrcxt;
	volatile xsltStylesheetPtr fc_stylesheet = NULL;
	volatile xmlDocPtr fc_doctree = NULL;
	volatile xmlDocPtr fc_restree = NULL;
	volatile xsltSecurityPrefsPtr fc_xslt_sec_prefs = NULL;
	volatile xsltTransformContextPtr fc_xslt_ctxt = NULL;
	volatile int fc_resstat = -1;
	xmlChar    *fc_resstr = NULL;
	int			fc_reslen = 0;

	if (fcinfo->nargs == 3)
	{
		fc_paramstr = PG_GETARG_TEXT_PP(2);
		fc_params = fc_parse_params(fc_paramstr);
	}
	else
	{
		/* 无参数 */
		fc_params = (const char **) palloc(sizeof(char *));
		fc_params[0] = NULL;
	}

	/* 设置解析器 */
	fc_xmlerrcxt = pgxml_parser_init(PG_XML_STRICTNESS_LEGACY);

	PG_TRY();
	{
		xmlDocPtr	fc_ssdoc;
		bool		fc_xslt_sec_prefs_error;

		/* 解析文档 */
		fc_doctree = xmlReadMemory((char *) VARDATA_ANY(fc_doct),
								VARSIZE_ANY_EXHDR(fc_doct), NULL, NULL,
								XML_PARSE_NOENT);

		if (fc_doctree == NULL)
			xml_ereport(fc_xmlerrcxt, ERROR, ERRCODE_EXTERNAL_ROUTINE_EXCEPTION,
						"error parsing XML document");

		/* 样式表同样 */
		fc_ssdoc = xmlReadMemory((char *) VARDATA_ANY(fc_ssheet),
							  VARSIZE_ANY_EXHDR(fc_ssheet), NULL, NULL,
							  XML_PARSE_NOENT);

		if (fc_ssdoc == NULL)
			xml_ereport(fc_xmlerrcxt, ERROR, ERRCODE_EXTERNAL_ROUTINE_EXCEPTION,
						"error parsing stylesheet as XML document");

		/* 在这个调用后，我们不需要单独释放 ssdoc */
		fc_stylesheet = xsltParseStylesheetDoc(fc_ssdoc);

		if (fc_stylesheet == NULL)
			xml_ereport(fc_xmlerrcxt, ERROR, ERRCODE_EXTERNAL_ROUTINE_EXCEPTION,
						"failed to parse stylesheet");

		fc_xslt_ctxt = xsltNewTransformContext(fc_stylesheet, fc_doctree);

		fc_xslt_sec_prefs_error = false;
		if ((fc_xslt_sec_prefs = xsltNewSecurityPrefs()) == NULL)
			fc_xslt_sec_prefs_error = true;

		if (xsltSetSecurityPrefs(fc_xslt_sec_prefs, XSLT_SECPREF_READ_FILE,
								 xsltSecurityForbid) != 0)
			fc_xslt_sec_prefs_error = true;
		if (xsltSetSecurityPrefs(fc_xslt_sec_prefs, XSLT_SECPREF_WRITE_FILE,
								 xsltSecurityForbid) != 0)
			fc_xslt_sec_prefs_error = true;
		if (xsltSetSecurityPrefs(fc_xslt_sec_prefs, XSLT_SECPREF_CREATE_DIRECTORY,
								 xsltSecurityForbid) != 0)
			fc_xslt_sec_prefs_error = true;
		if (xsltSetSecurityPrefs(fc_xslt_sec_prefs, XSLT_SECPREF_READ_NETWORK,
								 xsltSecurityForbid) != 0)
			fc_xslt_sec_prefs_error = true;
		if (xsltSetSecurityPrefs(fc_xslt_sec_prefs, XSLT_SECPREF_WRITE_NETWORK,
								 xsltSecurityForbid) != 0)
			fc_xslt_sec_prefs_error = true;
		if (xsltSetCtxtSecurityPrefs(fc_xslt_sec_prefs, fc_xslt_ctxt) != 0)
			fc_xslt_sec_prefs_error = true;

		if (fc_xslt_sec_prefs_error)
			ereport(ERROR,
					(errmsg("could not set libxslt security preferences")));

		fc_restree = xsltApplyStylesheetUser(fc_stylesheet, fc_doctree, fc_params,
										  NULL, NULL, fc_xslt_ctxt);

		if (fc_restree == NULL)
			xml_ereport(fc_xmlerrcxt, ERROR, ERRCODE_EXTERNAL_ROUTINE_EXCEPTION,
						"failed to apply stylesheet");

		fc_resstat = xsltSaveResultToString(&fc_resstr, &fc_reslen, fc_restree, fc_stylesheet);
	}
	PG_CATCH();
	{
		if (fc_restree != NULL)
			xmlFreeDoc(fc_restree);
		if (fc_xslt_ctxt != NULL)
			xsltFreeTransformContext(fc_xslt_ctxt);
		if (fc_xslt_sec_prefs != NULL)
			xsltFreeSecurityPrefs(fc_xslt_sec_prefs);
		if (fc_stylesheet != NULL)
			xsltFreeStylesheet(fc_stylesheet);
		if (fc_doctree != NULL)
			xmlFreeDoc(fc_doctree);
		xsltCleanupGlobals();

		pg_xml_done(fc_xmlerrcxt, true);

		PG_RE_THROW();
	}
	PG_END_TRY();

	xmlFreeDoc(fc_restree);
	xsltFreeTransformContext(fc_xslt_ctxt);
	xsltFreeSecurityPrefs(fc_xslt_sec_prefs);
	xsltFreeStylesheet(fc_stylesheet);
	xmlFreeDoc(fc_doctree);
	xsltCleanupGlobals();

	pg_xml_done(fc_xmlerrcxt, false);

	/* XXX 这真的很可疑，应该抛出错误 */
	if (fc_resstat < 0)
		PG_RETURN_NULL();

	fc_result = cstring_to_text_with_len((char *) fc_resstr, fc_reslen);

	if (fc_resstr)
		xmlFree(fc_resstr);

	PG_RETURN_TEXT_P(fc_result);
#else							/* !USE_LIBXSLT */

	ereport(ERROR,
			(errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
			 errmsg("xslt_process() is not available without libxslt")));
	PG_RETURN_NULL();
#endif							/* USE_LIBXSLT */
}

#ifdef USE_LIBXSLT

static const char **
fc_parse_params(text *fc_paramstr)
{
	char	   *fc_pos;
	char	   *fc_pstr;
	char	   *fc_nvsep = "=";
	char	   *fc_itsep = ",";
	const char **fc_params;
	int			fc_max_params;
	int			fc_nparams;

	fc_pstr = text_to_cstring(fc_paramstr);

	fc_max_params = 20;			/* 必须是偶数！ */
	fc_params = (const char **) palloc((fc_max_params + 1) * sizeof(char *));
	fc_nparams = 0;

	fc_pos = fc_pstr;

	while (*fc_pos != '\0')
	{
		if (fc_nparams >= fc_max_params)
		{
			fc_max_params *= 2;
			fc_params = (const char **) repalloc(fc_params,
											  (fc_max_params + 1) * sizeof(char *));
		}
		fc_params[fc_nparams++] = fc_pos;
		fc_pos = strstr(fc_pos, fc_nvsep);
		if (fc_pos != NULL)
		{
			*fc_pos = '\0';
			fc_pos++;
		}
		else
		{
			/* 没有等号，因此忽略这个“参数” */
			fc_nparams--;
			break;
		}

		/* 由于 max_params 是偶数，我们仍然有 nparams < max_params */
		fc_params[fc_nparams++] = fc_pos;
		fc_pos = strstr(fc_pos, fc_itsep);
		if (fc_pos != NULL)
		{
			*fc_pos = '\0';
			fc_pos++;
		}
		else
			break;
	}

	/* 添加终止符标记；我们在 palloc 中留出了空间 */
	fc_params[fc_nparams] = NULL;

	return fc_params;
}

#endif							/* USE_LIBXSLT */
